from pathlib import Path

BASE_DIR = Path(__file__).parent.parent
DATA_DIR = BASE_DIR / "data" / "datasets"
CHECKPOINT_DIR = BASE_DIR / "data" / "checkpoints"
CHECKPOINT_DIR.mkdir(parents=True, exist_ok=True)

EPOCHS = 10
LR = 0.001
BATCH_SIZE = 32